from ultralytics import YOLO

# yolo = YOLO("./yolov8n.pt", task="detect")
yolo = YOLO("./best.pt", task="detect")
# result = yolo(source="screen")
result = yolo(source="./ultralytics/assets/bus.jpg")
result[0].save("annotated_bus.jpg")  # 手动指定保存路径
